package org.ala.linshen.dp;

/**
 * 给你一个整数数组 nums ，你可以对它进行一些操作。
 *
 * 每次操作中，选择任意一个 nums[i] ，删除它并获得 nums[i] 的点数。之后，你必须删除 所有 等于 nums[i] - 1 和 nums[i] + 1 的元素。
 *
 * 开始你拥有 0 个点数。返回你能通过这些操作获得的最大点数。
 *
 *
 *
 * 示例 1：
 *
 * 输入：nums = [3,4,2]
 * 输出：6
 * 解释：
 * 删除 4 获得 4 个点数，因此 3 也被删除。
 * 之后，删除 2 获得 2 个点数。总共获得 6 个点数。
 * 示例 2：
 *
 * 输入：nums = [2,2,3,3,3,4]
 * 输出：9
 * 解释：
 * 删除 3 获得 3 个点数，接着要删除两个 2 和 4 。
 * 之后，再次删除 3 获得 3 个点数，再次删除 3 获得 3 个点数。
 * 总共获得 9 个点数。
 *
 *
 * 提示：
 *
 * 1 <= nums.length <= 2 * 104
 * 1 <= nums[i] <= 104
 *
 * @author ala
 * @data 2024-09-13 16:18
 */
public class Q740 {

    public static void main(String[] args) {
        Q740 q = new Q740();

//        int[] nums = {2,2,3,3,3,4};
        int[] nums = {1};

        System.out.println(q.deleteAndEarn(nums));
    }


    public int deleteAndEarn(int[] nums) {
        N = nums.length;
        return V1(nums);
    }
    protected int N;

    /**
     *  score[i]表示数字i的得分
     *      score[i] = nums[i] * cnts[i]
     *  dp[i]表示截止到数字i，能得到的得分
     *      dp[i] = max(dp[i - 2] + score[i], dp[i - 1])
     *      和打家劫舍一样，偷中间就不能偷两边，偷两边就不能偷中间
     */
    protected int V1(int[] nums) {
        int[] scores = new int[10001];
        int max = -1;
        for (int n : nums) {
            scores[n] += n;
            max = Math.max(max, n);
        }

        int[] dp = new int[max + 1];
        dp[1] = scores[1];
        for (int i = 2 ; i <= max ; i++) {
            dp[i] = Math.max(dp[i - 2] + scores[i], dp[i - 1]);
        }
        return dp[max];
    }
}
